import logging

class TrainingLogger:
    def __init__(self):
        logging.basicConfig(level=logging.INFO)

    def log_epoch(self, epoch, loss):
        logging.info(f"Epoch {epoch} Loss: {loss}")